package util

import "github.com/tmc/langchaingo/prompts"

// 创建提词器模板
func Prompt() prompts.ChatPromptTemplate {
	return prompts.NewChatPromptTemplate([]prompts.MessageFormatter{
		//系统
		prompts.NewSystemMessagePromptTemplate("", nil),
		//人类语言模板
		prompts.NewHumanMessagePromptTemplate("{{.text}}", []string{"text"}),
		// ai 提示模板
		prompts.NewAIMessagePromptTemplate("ai测试回复信息", nil),
	})
}
